package com.shengsiyuan.netty.thirdexample;

import io.netty.channel.Channel;
import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.SimpleChannelInboundHandler;
import io.netty.channel.group.ChannelGroup;
import io.netty.channel.group.DefaultChannelGroup;
import io.netty.util.concurrent.GlobalEventExecutor;

/**
 * Chat服务端的ServerHandler
 * 需求1：服务器首先启动，然后ABC三个客户端与服务器建立连接，当第一个客户端建立之后，什么也不发生。第二个又连接之后，
 * 之前已经建立好连接的客户端，服务器会打印，xxx已经上线，客户端失去连接的时候，服务端会打印xxx下线。客户端上线之后，
 * 还会通知其他的客户端。
 * 需求2：ABC建立好了连接之后，当A发送消息到服务器，ABC都会收到消息。
 *
 * @author 荀彧 2021/1/5
 */
public class MyChatServerHandler extends SimpleChannelInboundHandler<String> {

    /**
     * 保存channel 对象。所有与服务端建立连接的对象。
     */
    private static ChannelGroup channelGroup = new DefaultChannelGroup(GlobalEventExecutor.INSTANCE);

    /**
     * 服务器收到任何一个客户端发来的消息。
     * @param ctx
     * @param msg
     * @throws Exception
     */
    @Override
    protected void channelRead0(ChannelHandlerContext ctx, String msg) throws Exception {
        Channel channel = ctx.channel();
        channelGroup.forEach(ch -> {
            if (channel != ch) {
                ch.writeAndFlush(channel.remoteAddress() + "发送的消息：" + msg + "\n");
            }else {
                // 自己
                ch.writeAndFlush("[自己]" + msg + "\n");
            }
        });
    }

    /**
     * 客户端与服务端已经建立。
     * @param ctx
     * @throws Exception
     */
    @Override
    public void handlerAdded(ChannelHandlerContext ctx) throws Exception {
        Channel channel = ctx.channel();
        // 会向channelGroup中所有的对象发送消息。
        channelGroup.writeAndFlush("[服务器] --"+ channel.remoteAddress() + "加入\n");
        // 将其加入channelGroup
        channelGroup.add(channel);
    }

    /**
     * 客户端连接断掉之后
     * @param ctx
     * @throws Exception
     */
    @Override
    public void handlerRemoved(ChannelHandlerContext ctx) throws Exception {
        Channel channel = ctx.channel();
        channelGroup.writeAndFlush("[服务器] --" + channel.remoteAddress() + "断开\n");
        // 这个无需手动触发，netty会自动触发。
        // channelGroup.remove(channel);

    }

    /**
     * 表示连接处于活动状态。
     * @param ctx
     * @throws Exception
     */
    @Override
    public void channelActive(ChannelHandlerContext ctx) throws Exception {
        Channel channel = ctx.channel();
        System.out.println(channel.remoteAddress() + " 上线");
    }

    /**
     * 表示连接下线。
     * @param ctx
     * @throws Exception
     */
    @Override
    public void channelInactive(ChannelHandlerContext ctx) throws Exception {
        Channel channel = ctx.channel();
        System.out.println(channel.remoteAddress() + " 下线");
    }

    /**
     * 异常退出的情景
     * @param ctx
     * @param cause
     * @throws Exception
     */
    @Override
    public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) throws Exception {
        cause.printStackTrace();
        ctx.close();
    }
}

